package com.example.chamberlainserver.Service;

import com.example.chamberlainserver.Entry.Role;
import com.example.chamberlainserver.Entry.User;
import com.example.chamberlainserver.security.RoleConstants;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * 角色验证服务
 * 提供用户角色验证相关功能
 */
@Service
public class RoleValidationService {

    @Autowired
    private UserRoleService userRoleService;

    @Autowired
    private UserService userService;

    /**
     * 有效角色代码集合
     */
    private static final Set<String> VALID_ROLE_CODES = new HashSet<>(Arrays.asList(
            RoleConstants.ADMIN,       // 超级管理员
            RoleConstants.TEACHER,     // 辅导员
            RoleConstants.MINISTER,    // 部长
            RoleConstants.MEMBER       // 部员
    ));

    /**
     * 验证用户是否具有有效角色
     * 
     * @param userId 用户ID
     * @return 是否具有有效角色
     */
    public boolean hasValidRole(Integer userId) {
        // 获取用户角色列表
        List<Role> roles = userRoleService.getRolesByUserId(userId);
        
        // 如果用户没有关联角色，则返回false
        if (roles == null || roles.isEmpty()) {
            return false;
        }
        
        // 检查用户的角色是否在有效角色列表中
        return roles.stream()
                .map(Role::getRoleCode)
                .anyMatch(VALID_ROLE_CODES::contains);
    }
    
    /**
     * 获取用户的有效角色列表
     * 
     * @param userId 用户ID
     * @return 有效角色列表
     */
    public List<String> getValidRoles(Integer userId) {
        // 获取用户角色列表
        List<Role> roles = userRoleService.getRolesByUserId(userId);
        
        // 如果用户没有关联角色，则返回空列表
        if (roles == null || roles.isEmpty()) {
            return Arrays.asList();
        }
        
        // 过滤出有效角色
        return roles.stream()
                .map(Role::getRoleCode)
                .filter(VALID_ROLE_CODES::contains)
                .collect(Collectors.toList());
    }
    
    /**
     * 检查用户是否为超级管理员
     */
    public boolean isAdmin(Integer userId) {
        return hasRole(userId, RoleConstants.ADMIN);
    }
    
    /**
     * 检查用户是否为辅导员
     */
    public boolean isTeacher(Integer userId) {
        return hasRole(userId, RoleConstants.TEACHER);
    }
    
    /**
     * 检查用户是否为部长
     */
    public boolean isMinister(Integer userId) {
        return hasRole(userId, RoleConstants.MINISTER);
    }
    
    /**
     * 检查用户是否为部员
     */
    public boolean isMember(Integer userId) {
        return hasRole(userId, RoleConstants.MEMBER);
    }
    
    /**
     * 检查用户是否具有指定角色
     */
    private boolean hasRole(Integer userId, String roleCode) {
        // 获取用户角色列表
        List<Role> roles = userRoleService.getRolesByUserId(userId);
        
        // 如果用户没有关联角色，则返回false
        if (roles == null || roles.isEmpty()) {
            return false;
        }
        
        // 检查用户是否具有指定角色
        return roles.stream()
                .map(Role::getRoleCode)
                .anyMatch(code -> code.equals(roleCode));
    }
}